/*
 *
 * Paros and its related class files.
 * 
 * Paros is an HTTP/HTTPS proxy for assessing web application security.
 * Copyright (C) 2003-2004 Chinotec Technologies Company
 * 
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the Clarified Artistic License
 * as published by the Free Software Foundation.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * Clarified Artistic License for more details.
 * 
 * You should have received a copy of the Clarified Artistic License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */
package org.parosproxy.paros.view;

import java.awt.Component;

import javax.swing.JTree;
import javax.swing.tree.TreePath;

import org.parosproxy.paros.control.Control;
import org.parosproxy.paros.extension.ExtensionPopupMenu;
import org.parosproxy.paros.extension.history.ExtensionHistory;
import org.parosproxy.paros.model.HistoryReference;
import org.parosproxy.paros.model.SiteMap;
import org.parosproxy.paros.model.SiteNode;

/**
 * 
 * To change the template for this generated type comment go to Window -
 * Preferences - Java - Code Generation - Code and Comments
 */
public class PopupDeleteMenu extends ExtensionPopupMenu {

	private static final long serialVersionUID = 2282358266003940700L;
	
	private Component invoker = null;

	/**
	 * This method initializes
	 * 
	 */
	public PopupDeleteMenu() {
		super();
		initialize();
	}

	/**
	 * This method initializes this
	 * 
	 * @return void
	 */
	private void initialize() {
		this.setText("Delete (from view)");
		this.addActionListener(new java.awt.event.ActionListener() {

			public void actionPerformed(java.awt.event.ActionEvent e) {
				if (invoker.getName().equals("treeSite")) {
					JTree tree = (JTree) invoker;
					TreePath[] paths = tree.getSelectionPaths();
					SiteMap map = (SiteMap) tree.getModel();
					for (int i = 0; i < paths.length; i++) {
						SiteNode node = (SiteNode) paths[i]
								.getLastPathComponent();
						delete(map, node);
					}
				}

			}
		});

	}

	public boolean isEnableForComponent(Component invoker) {
		if (invoker.getName() != null && invoker.getName().equals("treeSite")) {
			this.invoker = invoker;
			return true;
		} else {
			this.invoker = null;
			return false;
		}

	}

	public static void delete(SiteMap map, SiteNode node) {
		SiteNode child = null;
		synchronized (map) {
			while (node.getChildCount() > 0) {
				try {
					child = (SiteNode) node.getChildAt(0);
					delete(map, child);
				} catch (Exception e) {
				}
			}

			if (node.isRoot()) {
				return;
			}

			// delete reference in node

			ExtensionHistory ext = (ExtensionHistory) Control.getSingleton()
					.getExtensionLoader().getExtension("ExtensionHistory");
			ext.getHistoryList().removeElement(node.getHistoryReference());

			// delete past reference in node
			while (node.getPastHistoryReference().size() > 0) {
				HistoryReference ref = (HistoryReference) node
						.getPastHistoryReference().get(0);
				ext.getHistoryList().removeElement(ref);
				node.getPastHistoryReference().remove(0);
			}

			map.removeNodeFromParent(node);
		}
	}

}
